/*
 * winutils.c: miscellaneous Windows utilities for GUI apps
 */

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>

#include "putty.h"
#include "misc.h"

#ifdef TESTMODE
/* Definitions to allow this module to be compiled standalone for testing
 * split_into_argv(). */
#define smalloc malloc
#define srealloc realloc
#define sfree free
#endif

/*
 * GetOpenFileName/GetSaveFileName tend to muck around with the process'
 * working directory on at least some versions of Windows.
 * Here's a wrapper that gives more control over this, and hides a little
 * bit of other grottiness.
 */

struct filereq_tag {
  TCHAR cwd[MAX_PATH];
};

/*
 * `of' is expected to be initialised with most interesting fields, but
 * this function does some administrivia. (assume `of' was memset to 0)
 * save==1 -> GetSaveFileName; save==0 -> GetOpenFileName
 * `state' is optional.
 */
BOOL request_file(filereq *state, OPENFILENAME *of, int preserve, int save)
{
  TCHAR cwd[MAX_PATH]; /* process CWD */
  BOOL ret;

  /* Get process CWD */
  if (preserve) {
    DWORD r = GetCurrentDirectory(lenof(cwd), cwd);
    if (r == 0 || r >= lenof(cwd))
      /* Didn't work, oh well. Stop trying to be clever. */
      preserve = 0;
  }

  /* Open the file requester, maybe setting lpstrInitialDir */
  {
#ifdef OPENFILENAME_SIZE_VERSION_400
    of->lStructSize = OPENFILENAME_SIZE_VERSION_400;
#else
    of->lStructSize = sizeof(*of);
#endif
    of->lpstrInitialDir = (state && state->cwd[0]) ? state->cwd : NULL;
    /* Actually put up the requester. */
    ret = save ? GetSaveFileName(of) : GetOpenFileName(of);
  }

  /* Get CWD left by requester */
  if (state) {
    DWORD r = GetCurrentDirectory(lenof(state->cwd), state->cwd);
    if (r == 0 || r >= lenof(state->cwd))
      /* Didn't work, oh well. */
      state->cwd[0] = '\0';
  }

  /* Restore process CWD */
  if (preserve)
    /* If it fails, there's not much we can do. */
    (void)SetCurrentDirectory(cwd);

  return ret;
}

filereq *filereq_new(void)
{
  filereq *ret = snew(filereq);
  ret->cwd[0] = '\0';
  return ret;
}

void filereq_free(filereq *state)
{
  sfree(state);
}

/*
 * Message box with optional context help.
 */

/* Callback function to launch context help. */
static VOID CALLBACK message_box_help_callback(LPHELPINFO lpHelpInfo)
{
  const char *context = NULL;
#define CHECK_CTX(name)                                                        \
  do {                                                                         \
    if (lpHelpInfo->dwContextId == WINHELP_CTXID_##name)                       \
      context = WINHELP_CTX_##name;                                            \
  } while (0)
  CHECK_CTX(errors_hostkey_absent);
  CHECK_CTX(errors_hostkey_changed);
  CHECK_CTX(errors_cantloadkey);
  CHECK_CTX(option_cleanup);
  CHECK_CTX(pgp_fingerprints);
#undef CHECK_CTX
  if (context)
    launch_help(hwnd, context);
}

int message_box(LPCTSTR text, LPCTSTR caption, DWORD style, DWORD helpctxid)
{
  MSGBOXPARAMS mbox;

  /*
   * We use MessageBoxIndirect() because it allows us to specify a
   * callback function for the Help button.
   */
  mbox.cbSize = sizeof(mbox);
  /* Assumes the globals `hinst' and `hwnd' have sensible values. */
  mbox.hInstance = hinst;
  mbox.hwndOwner = hwnd;
  mbox.lpfnMsgBoxCallback = &message_box_help_callback;
  mbox.dwLanguageId = LANG_NEUTRAL;
  mbox.lpszText = text;
  mbox.lpszCaption = caption;
  mbox.dwContextHelpId = helpctxid;
  mbox.dwStyle = style;
  if (helpctxid != 0 && has_help())
    mbox.dwStyle |= MB_HELP;
  return MessageBoxIndirect(&mbox);
}

/*
 * Display the fingerprints of the PGP Master Keys to the user.
 */
void pgp_fingerprints(void)
{
  message_box(
      "These are the fingerprints of the PuTTY PGP Master Keys. They can\n"
      "be used to establish a trust path from this executable to another\n"
      "one. See the manual for more information.\n"
      "(Note: these fingerprints have nothing to do with SSH!)\n"
      "\n"
      "PuTTY Master Key as of 2015 (RSA, 4096-bit):\n"
      "  " PGP_MASTER_KEY_FP "\n\n"
      "Original PuTTY Master Key (RSA, 1024-bit):\n"
      "  " PGP_RSA_MASTER_KEY_FP "\n"
      "Original PuTTY Master Key (DSA, 1024-bit):\n"
      "  " PGP_DSA_MASTER_KEY_FP,
      "PGP fingerprints",
      MB_ICONINFORMATION | MB_OK,
      HELPCTXID(pgp_fingerprints));
}

/*
 * Handy wrapper around GetDlgItemText which doesn't make you invent
 * an arbitrary length limit on the output string. Returned string is
 * dynamically allocated; caller must free.
 */
char *GetDlgItemText_alloc(HWND hwnd, int id)
{
  char *ret = NULL;
  int size = 0;

  do {
    size = size * 4 / 3 + 512;
    ret = sresize(ret, size, char);
    GetDlgItemText(hwnd, id, ret, size);
  } while (!memchr(ret, '\0', size - 1));

  return ret;
}

/*
 * Split a complete command line into argc/argv, attempting to do it
 * exactly the same way the Visual Studio C library would do it (so
 * that our console utilities, which receive argc and argv already
 * broken apart by the C library, will have their command lines
 * processed in the same way as the GUI utilities which get a whole
 * command line and must call this function).
 *
 * Does not modify the input command line.
 *
 * The final parameter (argstart) is used to return a second array
 * of char * pointers, the same length as argv, each one pointing
 * at the start of the corresponding element of argv in the
 * original command line. So if you get half way through processing
 * your command line in argc/argv form and then decide you want to
 * treat the rest as a raw string, you can. If you don't want to,
 * `argstart' can be safely left NULL.
 */
void split_into_argv(char *cmdline, int *argc, char ***argv, char ***argstart)
{
  char *p;
  char *outputline, *q;
  char **outputargv, **outputargstart;
  int outputargc;

  /*
   * These argument-breaking rules apply to Visual Studio 7, which
   * is currently the compiler expected to be used for PuTTY. Visual
   * Studio 10 has different rules, lacking the curious mod 3
   * behaviour of consecutive quotes described below; I presume they
   * fixed a bug. As and when we migrate to a newer compiler, we'll
   * have to adjust this to match; however, for the moment we
   * faithfully imitate in our GUI utilities what our CLI utilities
   * can't be prevented from doing.
   *
   * When I investigated this, at first glance the rules appeared to
   * be:
   *
   *  - Single quotes are not special characters.
   *
   *  - Double quotes are removed, but within them spaces cease
   *    to be special.
   *
   *  - Backslashes are _only_ special when a sequence of them
   *    appear just before a double quote. In this situation,
   *    they are treated like C backslashes: so \" just gives a
   *    literal quote, \\" gives a literal backslash and then
   *    opens or closes a double-quoted segment, \\\" gives a
   *    literal backslash and then a literal quote, \\\\" gives
   *    two literal backslashes and then opens/closes a
   *    double-quoted segment, and so forth. Note that this
   *    behaviour is identical inside and outside double quotes.
   *
   *  - Two successive double quotes become one literal double
   *    quote, but only _inside_ a double-quoted segment.
   *    Outside, they just form an empty double-quoted segment
   *    (which may cause an empty argument word).
   *
   *  - That only leaves the interesting question of what happens
   *    when one or more backslashes precedes two or more double
   *    quotes, starting inside a double-quoted string. And the
   *    answer to that appears somewhat bizarre. Here I tabulate
   *    number of backslashes (across the top) against number of
   *    quotes (down the left), and indicate how many backslashes
   *    are output, how many quotes are output, and whether a
   *    quoted segment is open at the end of the sequence:
   *
   *                      backslashes
   *
   *               0         1      2      3      4
   *
   *         0   0,0,y  |  1,0,y  2,0,y  3,0,y  4,0,y
   *            --------+-----------------------------
   *         1   0,0,n  |  0,1,y  1,0,n  1,1,y  2,0,n
   *    q    2   0,1,n  |  0,1,n  1,1,n  1,1,n  2,1,n
   *    u    3   0,1,y  |  0,2,n  1,1,y  1,2,n  2,1,y
   *    o    4   0,1,n  |  0,2,y  1,1,n  1,2,y  2,1,n
   *    t    5   0,2,n  |  0,2,n  1,2,n  1,2,n  2,2,n
   *    e    6   0,2,y  |  0,3,n  1,2,y  1,3,n  2,2,y
   *    s    7   0,2,n  |  0,3,y  1,2,n  1,3,y  2,2,n
   *         8   0,3,n  |  0,3,n  1,3,n  1,3,n  2,3,n
   *         9   0,3,y  |  0,4,n  1,3,y  1,4,n  2,3,y
   *        10   0,3,n  |  0,4,y  1,3,n  1,4,y  2,3,n
   *        11   0,4,n  |  0,4,n  1,4,n  1,4,n  2,4,n
   *
   *
   *      [Test fragment was of the form "a\\\"""b c" d.]
   *
   * There is very weird mod-3 behaviour going on here in the
   * number of quotes, and it even applies when there aren't any
   * backslashes! How ghastly.
   *
   * With a bit of thought, this extremely odd diagram suddenly
   * coalesced itself into a coherent, if still ghastly, model of
   * how things work:
   *
   *  - As before, backslashes are only special when one or more
   *    of them appear contiguously before at least one double
   *    quote. In this situation the backslashes do exactly what
   *    you'd expect: each one quotes the next thing in front of
   *    it, so you end up with n/2 literal backslashes (if n is
   *    even) or (n-1)/2 literal backslashes and a literal quote
   *    (if n is odd). In the latter case the double quote
   *    character right after the backslashes is used up.
   *
   *  - After that, any remaining double quotes are processed. A
   *    string of contiguous unescaped double quotes has a mod-3
   *    behaviour:
   *
   *     * inside a quoted segment, a quote ends the segment.
   *     * _immediately_ after ending a quoted segment, a quote
   *       simply produces a literal quote.
   *     * otherwise, outside a quoted segment, a quote begins a
   *       quoted segment.
   *
   *    So, for example, if we started inside a quoted segment
   *    then two contiguous quotes would close the segment and
   *    produce a literal quote; three would close the segment,
   *    produce a literal quote, and open a new segment. If we
   *    started outside a quoted segment, then two contiguous
   *    quotes would open and then close a segment, producing no
   *    output (but potentially creating a zero-length argument);
   *    but three quotes would open and close a segment and then
   *    produce a literal quote.
   */

  /*
   * First deal with the simplest of all special cases: if there
   * aren't any arguments, return 0,NULL,NULL.
   */
  while (*cmdline && isspace(*cmdline))
    cmdline++;
  if (!*cmdline) {
    if (argc)
      *argc = 0;
    if (argv)
      *argv = NULL;
    if (argstart)
      *argstart = NULL;
    return;
  }

  /*
   * This will guaranteeably be big enough; we can realloc it
   * down later.
   */
  outputline = snewn(1 + strlen(cmdline), char);
  outputargv = snewn(strlen(cmdline) + 1 / 2, char *);
  outputargstart = snewn(strlen(cmdline) + 1 / 2, char *);

  p = cmdline;
  q = outputline;
  outputargc = 0;

  while (*p) {
    int quote;

    /* Skip whitespace searching for start of argument. */
    while (*p && isspace(*p))
      p++;
    if (!*p)
      break;

    /* We have an argument; start it. */
    outputargv[outputargc] = q;
    outputargstart[outputargc] = p;
    outputargc++;
    quote = 0;

    /* Copy data into the argument until it's finished. */
    while (*p) {
      if (!quote && isspace(*p))
        break; /* argument is finished */

      if (*p == '"' || *p == '\\') {
        /*
         * We have a sequence of zero or more backslashes
         * followed by a sequence of zero or more quotes.
         * Count up how many of each, and then deal with
         * them as appropriate.
         */
        int i, slashes = 0, quotes = 0;
        while (*p == '\\')
          slashes++, p++;
        while (*p == '"')
          quotes++, p++;

        if (!quotes) {
          /*
           * Special case: if there are no quotes,
           * slashes are not special at all, so just copy
           * n slashes to the output string.
           */
          while (slashes--)
            *q++ = '\\';
        } else {
          /* Slashes annihilate in pairs. */
          while (slashes >= 2)
            slashes -= 2, *q++ = '\\';

          /* One remaining slash takes out the first quote. */
          if (slashes)
            quotes--, *q++ = '"';

          if (quotes > 0) {
            /* Outside a quote segment, a quote starts one. */
            if (!quote)
              quotes--;

            /* Now we produce (n+1)/3 literal quotes... */
            for (i = 3; i <= quotes + 1; i += 3)
              *q++ = '"';

            /* ... and end in a quote segment iff 3 divides n. */
            quote = (quotes % 3 == 0);
          }
        }
      } else {
        *q++ = *p++;
      }
    }

    /* At the end of an argument, just append a trailing NUL. */
    *q++ = '\0';
  }

  outputargv = sresize(outputargv, outputargc, char *);
  outputargstart = sresize(outputargstart, outputargc, char *);

  if (argc)
    *argc = outputargc;
  if (argv)
    *argv = outputargv;
  else
    sfree(outputargv);
  if (argstart)
    *argstart = outputargstart;
  else
    sfree(outputargstart);
}

#ifdef TESTMODE

const struct argv_test {
  const char *cmdline;
  const char *argv[10];
} argv_tests[] = {
    /*
     * We generate this set of tests by invoking ourself with
     * `-generate'.
     */
    {"ab c\" d", {"ab", "c d", NULL}},
    {"a\"b c\" d", {"ab c", "d", NULL}},
    {"a\"\"b c\" d", {"ab", "c d", NULL}},
    {"a\"\"\"b c\" d", {"a\"b", "c d", NULL}},
    {"a\"\"\"\"b c\" d", {"a\"b c", "d", NULL}},
    {"a\"\"\"\"\"b c\" d", {"a\"b", "c d", NULL}},
    {"a\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"a\"\"\"\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
    {"a\"\"\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"a\\b c\" d", {"a\\b", "c d", NULL}},
    {"a\\\"b c\" d", {"a\"b", "c d", NULL}},
    {"a\\\"\"b c\" d", {"a\"b c", "d", NULL}},
    {"a\\\"\"\"b c\" d", {"a\"b", "c d", NULL}},
    {"a\\\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"a\\\"\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
    {"a\\\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"a\\\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
    {"a\\\"\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b c", "d", NULL}},
    {"a\\\\b c\" d", {"a\\\\b", "c d", NULL}},
    {"a\\\\\"b c\" d", {"a\\b c", "d", NULL}},
    {"a\\\\\"\"b c\" d", {"a\\b", "c d", NULL}},
    {"a\\\\\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"a\\\\\"\"\"\"b c\" d", {"a\\\"b c", "d", NULL}},
    {"a\\\\\"\"\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"a\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"a\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
    {"a\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"a\\\\\\b c\" d", {"a\\\\\\b", "c d", NULL}},
    {"a\\\\\\\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"a\\\\\\\"\"b c\" d", {"a\\\"b c", "d", NULL}},
    {"a\\\\\\\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"a\\\\\\\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"a\\\\\\\"\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
    {"a\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"a\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
    {"a\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b c", "d", NULL}},
    {"a\\\\\\\\b c\" d", {"a\\\\\\\\b", "c d", NULL}},
    {"a\\\\\\\\\"b c\" d", {"a\\\\b c", "d", NULL}},
    {"a\\\\\\\\\"\"b c\" d", {"a\\\\b", "c d", NULL}},
    {"a\\\\\\\\\"\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
    {"a\\\\\\\\\"\"\"\"b c\" d", {"a\\\\\"b c", "d", NULL}},
    {"a\\\\\\\\\"\"\"\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
    {"a\\\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
    {"a\\\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b c", "d", NULL}},
    {"a\\\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
    {"\"ab c\" d", {"ab c", "d", NULL}},
    {"\"a\"b c\" d", {"ab", "c d", NULL}},
    {"\"a\"\"b c\" d", {"a\"b", "c d", NULL}},
    {"\"a\"\"\"b c\" d", {"a\"b c", "d", NULL}},
    {"\"a\"\"\"\"b c\" d", {"a\"b", "c d", NULL}},
    {"\"a\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"\"a\"\"\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
    {"\"a\"\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"\"a\"\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
    {"\"a\\b c\" d", {"a\\b c", "d", NULL}},
    {"\"a\\\"b c\" d", {"a\"b c", "d", NULL}},
    {"\"a\\\"\"b c\" d", {"a\"b", "c d", NULL}},
    {"\"a\\\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"\"a\\\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
    {"\"a\\\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
    {"\"a\\\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
    {"\"a\\\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b c", "d", NULL}},
    {"\"a\\\"\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
    {"\"a\\\\b c\" d", {"a\\\\b c", "d", NULL}},
    {"\"a\\\\\"b c\" d", {"a\\b", "c d", NULL}},
    {"\"a\\\\\"\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"\"a\\\\\"\"\"b c\" d", {"a\\\"b c", "d", NULL}},
    {"\"a\\\\\"\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"\"a\\\\\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"\"a\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
    {"\"a\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"\"a\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
    {"\"a\\\\\\b c\" d", {"a\\\\\\b c", "d", NULL}},
    {"\"a\\\\\\\"b c\" d", {"a\\\"b c", "d", NULL}},
    {"\"a\\\\\\\"\"b c\" d", {"a\\\"b", "c d", NULL}},
    {"\"a\\\\\\\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"\"a\\\\\\\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
    {"\"a\\\\\\\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
    {"\"a\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
    {"\"a\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b c", "d", NULL}},
    {"\"a\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
    {"\"a\\\\\\\\b c\" d", {"a\\\\\\\\b c", "d", NULL}},
    {"\"a\\\\\\\\\"b c\" d", {"a\\\\b", "c d", NULL}},
    {"\"a\\\\\\\\\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
    {"\"a\\\\\\\\\"\"\"b c\" d", {"a\\\\\"b c", "d", NULL}},
    {"\"a\\\\\\\\\"\"\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
    {"\"a\\\\\\\\\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
    {"\"a\\\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b c", "d", NULL}},
    {"\"a\\\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
    {"\"a\\\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"\"b", "c d", NULL}},
};

int main(int argc, char **argv)
{
  int i, j;

  if (argc > 1) {
    /*
     * Generation of tests.
     *
     * Given `-splat <args>', we print out a C-style
     * representation of each argument (in the form "a", "b",
     * NULL), backslash-escaping each backslash and double
     * quote.
     *
     * Given `-split <string>', we first doctor `string' by
     * turning forward slashes into backslashes, single quotes
     * into double quotes and underscores into spaces; and then
     * we feed the resulting string to ourself with `-splat'.
     *
     * Given `-generate', we concoct a variety of fun test
     * cases, encode them in quote-safe form (mapping \, " and
     * space to /, ' and _ respectively) and feed each one to
     * `-split'.
     */
    if (!strcmp(argv[1], "-splat")) {
      int i;
      char *p;
      for (i = 2; i < argc; i++) {
        putchar('"');
        for (p = argv[i]; *p; p++) {
          if (*p == '\\' || *p == '"')
            putchar('\\');
          putchar(*p);
        }
        printf("\", ");
      }
      printf("NULL");
      return 0;
    }

    if (!strcmp(argv[1], "-split") && argc > 2) {
      char *str = malloc(20 + strlen(argv[0]) + strlen(argv[2]));
      char *p, *q;

      q = str + sprintf(str, "%s -splat ", argv[0]);
      printf("    {\"");
      for (p = argv[2]; *p; p++, q++) {
        switch (*p) {
        case '/':
          printf("\\\\");
          *q = '\\';
          break;
        case '\'':
          printf("\\\"");
          *q = '"';
          break;
        case '_':
          printf(" ");
          *q = ' ';
          break;
        default:
          putchar(*p);
          *q = *p;
          break;
        }
      }
      *p = '\0';
      printf("\", {");
      fflush(stdout);

      system(str);

      printf("}},\n");

      return 0;
    }

    if (!strcmp(argv[1], "-generate")) {
      char *teststr, *p;
      int i, initialquote, backslashes, quotes;

      teststr = malloc(200 + strlen(argv[0]));

      for (initialquote = 0; initialquote <= 1; initialquote++) {
        for (backslashes = 0; backslashes < 5; backslashes++) {
          for (quotes = 0; quotes < 9; quotes++) {
            p = teststr + sprintf(teststr, "%s -split ", argv[0]);
            if (initialquote)
              *p++ = '\'';
            *p++ = 'a';
            for (i = 0; i < backslashes; i++)
              *p++ = '/';
            for (i = 0; i < quotes; i++)
              *p++ = '\'';
            *p++ = 'b';
            *p++ = '_';
            *p++ = 'c';
            *p++ = '\'';
            *p++ = '_';
            *p++ = 'd';
            *p = '\0';

            system(teststr);
          }
        }
      }
      return 0;
    }

    fprintf(stderr, "unrecognised option: \"%s\"\n", argv[1]);
    return 1;
  }

  /*
   * If we get here, we were invoked with no arguments, so just
   * run the tests.
   */

  for (i = 0; i < lenof(argv_tests); i++) {
    int ac;
    char **av;

    split_into_argv(argv_tests[i].cmdline, &ac, &av);

    for (j = 0; j < ac && argv_tests[i].argv[j]; j++) {
      if (strcmp(av[j], argv_tests[i].argv[j])) {
        printf("failed test %d (|%s|) arg %d: |%s| should be |%s|\n",
               i,
               argv_tests[i].cmdline,
               j,
               av[j],
               argv_tests[i].argv[j]);
      }
#ifdef VERBOSE
      else {
        printf("test %d (|%s|) arg %d: |%s| == |%s|\n",
               i,
               argv_tests[i].cmdline,
               j,
               av[j],
               argv_tests[i].argv[j]);
      }
#endif
    }
    if (j < ac)
      printf("failed test %d (|%s|): %d args returned, should be %d\n",
             i,
             argv_tests[i].cmdline,
             ac,
             j);
    if (argv_tests[i].argv[j])
      printf("failed test %d (|%s|): %d args returned, should be more\n",
             i,
             argv_tests[i].cmdline,
             ac);
  }

  return 0;
}

#endif
